FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel

RUN apt update -y \
    && apt install -y vim git

RUN apt-get install -y -f libglib2.0-0 libsm6 libxrender1 libxext6 libgl1-mesa-glx \
    && pip install opencv-python

RUN pip install easycore pycocotools tqdm
